package org.example.day;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * @Description: TODO
 * @Author wyatt
 * @Data 2024/05/31 10:29
 */
//给你一个下标从 0 开始的二维整数矩阵 grid，大小为 n * n ，其中的值在 [1, n²] 范围内。除了 a 出现 两次，b 缺失 之外，每个整数都
// 恰好出现一次 。
//
// 任务是找出重复的数字a 和缺失的数字 b 。
//
// 返回一个下标从 0 开始、长度为 2 的整数数组 ans ，其中 ans[0] 等于 a ，ans[1] 等于 b 。
//
//
//
// 示例 1：
//
//
//输入：grid = [[1,3],[2,2]]
//输出：[2,4]
//解释：数字 2 重复，数字 4 缺失，所以答案是 [2,4] 。
//
//
// 示例 2：
//
//
//输入：grid = [[9,1,7],[8,9,2],[3,4,6]]
//输出：[9,5]
//解释：数字 9 重复，数字 5 缺失，所以答案是 [9,5] 。
//
//
//
//
// 提示：
//
//
// 2 <= n == grid.length == grid[i].length <= 50
// 1 <= grid[i][j] <= n * n
// 对于所有满足1 <= x <= n * n 的 x ，恰好存在一个 x 与矩阵中的任何成员都不相等。
// 对于所有满足1 <= x <= n * n 的 x ，恰好存在一个 x 与矩阵中的两个成员相等。
// 除上述的两个之外，对于所有满足1 <= x <= n * n 的 x ，都恰好存在一对 i, j 满足 0 <= i, j <= n - 1 且
//grid[i][j] == x 。
//
//
// Related Topics 数组 哈希表 数学 矩阵 👍 10 👎 0

public class Solution2965 {

    public static void main(String[] args) {
        Solution2965 solution2965 = new Solution2965();
        int[] res = solution2965.findMissingAndRepeatedValues2(new int[][]{{1,3},{2,2}});

        for (int i = 0; i < res.length; i++) {
            System.out.println(res[i]);
        }

    }


    public int[] findMissingAndRepeatedValues(int[][] grid) {
        int[] res = new int[2];
        int repeatNum = 0;
        int lackNum = 0;
        int n = grid.length;

        HashMap<Integer, Integer> countMap = new HashMap<>();
        List<Integer> numList = new ArrayList<>();
        for(int i=1;i<=n*n;i++){
            numList.add(i);
        }

        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[i].length; j++) {
                if(countMap.containsKey(grid[i][j])){
                    repeatNum = grid[i][j];
                }
                countMap.putIfAbsent(grid[i][j], 1);
                numList.remove(Integer.valueOf(grid[i][j]));
            }
        }

        if(numList.size() == 1){
            lackNum = numList.get(0);
        }

        res[0] = repeatNum;
        res[1] = lackNum;
        return res;
    }

    public int[] findMissingAndRepeatedValues2(int[][] grid) {
        int[] res = new int[2];
        int repeatNum = 0;
        int lackNum = 0;
        int n = grid.length;

        int[] nums = new int[n*n];
        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[i].length; j++) {
                nums[j+i*n] = grid[i][j];
            }
        }

        Arrays.sort(nums);
        int index = 0;
        for (int i = 0; i < nums.length; i++) {

            if(lackNum == 0 && nums[i] != index+1){
                lackNum = index+1;
            }
            index++;

            if(repeatNum == 0 && i < nums.length-1 && nums[i] == nums[i+1]){
                repeatNum = nums[i];
                index--;
            }
        }

        res[0] = repeatNum;
        res[1] = lackNum == 0 ? n*n : lackNum;

        return res;
    }

}
